import pandas as pd
import matplotlib.pyplot as plt
from matplotlib.font_manager import FontProperties

# 设置中文字体
font = FontProperties(fname='C:/Windows/Fonts/simhei.ttf', size=12)

# 读取Excel文件
df = pd.read_excel('e:/组队你组不组/team-up-do-you-want-to-group1/project/练习一/FhjlViewDD.xlsx', engine='openpyxl')

# 使用'创建时间'作为日期列
df = df.rename(columns={'创建时间': '日期'})

# 筛选6月份数据
df['日期'] = pd.to_datetime(df['日期'])
june_data = df[df['日期'].dt.month == 6]

# 按车牌号分组并计算总货运量（使用'净重'或'毛重'作为货运量）
plate_totals = june_data.groupby('车辆')['净重'].sum().sort_values(ascending=False)

# 输出统计结果
print("6月份各车牌号总货运量(降序):")
print(plate_totals)

# 保存为CSV
plate_totals.to_csv('plate_stats_june.csv')
print("结果已保存为 plate_stats_june.csv")

# 绘制柱状图
plt.figure(figsize=(12, 6))
plate_totals.plot.bar()
# 修改图表标题和标签的字体设置
plt.title('6月份各车牌号总货运量', fontproperties=font)
plt.xlabel('车牌号', fontproperties=font)
plt.ylabel('总货运量', fontproperties=font)
plt.xticks(rotation=45, fontproperties=font)
plt.tight_layout()
plt.savefig('plate_stats_chart.png')
plt.show()
print("图表已保存为 plate_stats_chart.png")
print("脚本开始执行")